home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_repr.py < prev    next >
Text File  |  2005-10-18  |  12KB  |  305 lines
  1. """
  2.   Test cases for the repr module
  3.   Nick Mathewson
  4. """
  5.  
  6. import sys
  7. import os
  8. import unittest
  9.  
  10. from test.test_support import run_unittest
  11. from repr import repr as r # Don't shadow builtin repr
  12.  
  13.  
  14. def nestedTuple(nesting):
  15.     t = ()
  16.     for i in range(nesting):
  17.         t = (t,)
  18.     return t
  19.  
  20. class ReprTests(unittest.TestCase):
  21.  
  22.     def test_string(self):
  23.         eq = self.assertEquals
  24.         eq(r("abc"), "'abc'")
  25.         eq(r("abcdefghijklmnop"),"'abcdefghijklmnop'")
  26.  
  27.         s = "a"*30+"b"*30
  28.         expected = repr(s)[:13] + "..." + repr(s)[-14:]
  29.         eq(r(s), expected)
  30.  
  31.         eq(r("\"'"), repr("\"'"))
  32.         s = "\""*30+"'"*100
  33.         expected = repr(s)[:13] + "..." + repr(s)[-14:]
  34.         eq(r(s), expected)
  35.  
  36.     def test_container(self):
  37.         from array import array
  38.         from collections import deque
  39.  
  40.         eq = self.assertEquals
  41.         # Tuples give up after 6 elements
  42.         eq(r(()), "()")
  43.         eq(r((1,)), "(1,)")
  44.         eq(r((1, 2, 3)), "(1, 2, 3)")
  45.         eq(r((1, 2, 3, 4, 5, 6)), "(1, 2, 3, 4, 5, 6)")
  46.         eq(r((1, 2, 3, 4, 5, 6, 7)), "(1, 2, 3, 4, 5, 6, ...)")
  47.  
  48.         # Lists give up after 6 as well
  49.         eq(r([]), "[]")
  50.         eq(r([1]), "[1]")
  51.         eq(r([1, 2, 3]), "[1, 2, 3]")
  52.         eq(r([1, 2, 3, 4, 5, 6]), "[1, 2, 3, 4, 5, 6]")
  53.         eq(r([1, 2, 3, 4, 5, 6, 7]), "[1, 2, 3, 4, 5, 6, ...]")
  54.  
  55.         # Sets give up after 6 as well
  56.         eq(r(set([])), "set([])")
  57.         eq(r(set([1])), "set([1])")
  58.         eq(r(set([1, 2, 3])), "set([1, 2, 3])")
  59.         eq(r(set([1, 2, 3, 4, 5, 6])), "set([1, 2, 3, 4, 5, 6])")
  60.         eq(r(set([1, 2, 3, 4, 5, 6, 7])), "set([1, 2, 3, 4, 5, 6, ...])")
  61.  
  62.         # Frozensets give up after 6 as well
  63.         eq(r(frozenset([])), "frozenset([])")
  64.         eq(r(frozenset([1])), "frozenset([1])")
  65.         eq(r(frozenset([1, 2, 3])), "frozenset([1, 2, 3])")
  66.         eq(r(frozenset([1, 2, 3, 4, 5, 6])), "frozenset([1, 2, 3, 4, 5, 6])")
  67.         eq(r(frozenset([1, 2, 3, 4, 5, 6, 7])), "frozenset([1, 2, 3, 4, 5, 6, ...])")
  68.  
  69.         # collections.deque after 6
  70.         eq(r(deque([1, 2, 3, 4, 5, 6, 7])), "deque([1, 2, 3, 4, 5, 6, ...])")
  71.  
  72.         # Dictionaries give up after 4.
  73.         eq(r({}), "{}")
  74.         d = {'alice': 1, 'bob': 2, 'charles': 3, 'dave': 4}
  75.         eq(r(d), "{'alice': 1, 'bob': 2, 'charles': 3, 'dave': 4}")
  76.         d['arthur'] = 1
  77.         eq(r(d), "{'alice': 1, 'arthur': 1, 'bob': 2, 'charles': 3, ...}")
  78.  
  79.         # array.array after 5.
  80.         eq(r(array('i')), "array('i', [])")
  81.         eq(r(array('i', [1])), "array('i', [1])")
  82.         eq(r(array('i', [1, 2])), "array('i', [1, 2])")
  83.         eq(r(array('i', [1, 2, 3])), "array('i', [1, 2, 3])")
  84.         eq(r(array('i', [1, 2, 3, 4])), "array('i', [1, 2, 3, 4])")
  85.         eq(r(array('i', [1, 2, 3, 4, 5])), "array('i', [1, 2, 3, 4, 5])")
  86.         eq(r(array('i', [1, 2, 3, 4, 5, 6])),
  87.                    "array('i', [1, 2, 3, 4, 5, ...])")
  88.  
  89.     def test_numbers(self):
  90.         eq = self.assertEquals
  91.         eq(r(123), repr(123))
  92.         eq(r(123L), repr(123L))
  93.         eq(r(1.0/3), repr(1.0/3))
  94.  
  95.         n = 10L**100
  96.         expected = repr(n)[:18] + "..." + repr(n)[-19:]
  97.         eq(r(n), expected)
  98.  
  99.     def test_instance(self):
  100.         eq = self.assertEquals
  101.         i1 = ClassWithRepr("a")
  102.         eq(r(i1), repr(i1))
  103.  
  104.         i2 = ClassWithRepr("x"*1000)
  105.         expected = repr(i2)[:13] + "..." + repr(i2)[-14:]
  106.         eq(r(i2), expected)
  107.  
  108.         i3 = ClassWithFailingRepr()
  109.         eq(r(i3), ("<ClassWithFailingRepr instance at %x>"%id(i3)))
  110.  
  111.         s = r(ClassWithFailingRepr)
  112.         self.failUnless(s.startswith("<class "))
  113.         self.failUnless(s.endswith(">"))
  114.         self.failUnless(s.find("...") == 8)
  115.  
  116.     def test_file(self):
  117.         fp = open(unittest.__file__)
  118.         self.failUnless(repr(fp).startswith(
  119.             "<open file '%s', mode 'r' at 0x" % unittest.__file__))
  120.         fp.close()
  121.         self.failUnless(repr(fp).startswith(
  122.             "<closed file '%s', mode 'r' at 0x" % unittest.__file__))
  123.  
  124.     def test_lambda(self):
  125.         self.failUnless(repr(lambda x: x).startswith(
  126.             "<function <lambda"))
  127.         # XXX anonymous functions?  see func_repr
  128.  
  129.     def test_builtin_function(self):
  130.         eq = self.assertEquals
  131.         # Functions
  132.         eq(repr(hash), '<built-in function hash>')
  133.         # Methods
  134.         self.failUnless(repr(''.split).startswith(
  135.             '<built-in method split of str object at 0x'))
  136.  
  137.     def test_xrange(self):
  138.         import warnings
  139.         eq = self.assertEquals
  140.         eq(repr(xrange(1)), 'xrange(1)')
  141.         eq(repr(xrange(1, 2)), 'xrange(1, 2)')
  142.         eq(repr(xrange(1, 2, 3)), 'xrange(1, 4, 3)')
  143.  
  144.     def test_nesting(self):
  145.         eq = self.assertEquals
  146.         # everything is meant to give up after 6 levels.
  147.         eq(r([[[[[[[]]]]]]]), "[[[[[[[]]]]]]]")
  148.         eq(r([[[[[[[[]]]]]]]]), "[[[[[[[...]]]]]]]")
  149.  
  150.         eq(r(nestedTuple(6)), "(((((((),),),),),),)")
  151.         eq(r(nestedTuple(7)), "(((((((...),),),),),),)")
  152.  
  153.         eq(r({ nestedTuple(5) : nestedTuple(5) }),
  154.            "{((((((),),),),),): ((((((),),),),),)}")
  155.         eq(r({ nestedTuple(6) : nestedTuple(6) }),
  156.            "{((((((...),),),),),): ((((((...),),),),),)}")
  157.  
  158.         eq(r([[[[[[{}]]]]]]), "[[[[[[{}]]]]]]")
  159.         eq(r([[[[[[[{}]]]]]]]), "[[[[[[[...]]]]]]]")
  160.  
  161.     def test_buffer(self):
  162.         # XXX doesn't test buffers with no b_base or read-write buffers (see
  163.         # bufferobject.c).  The test is fairly incomplete too.  Sigh.
  164.         x = buffer('foo')
  165.         self.failUnless(repr(x).startswith('<read-only buffer for 0x'))
  166.  
  167.     def test_cell(self):
  168.         # XXX Hmm? How to get at a cell object?
  169.         pass
  170.  
  171.     def test_descriptors(self):
  172.         eq = self.assertEquals
  173.         # method descriptors
  174.         eq(repr(dict.items), "<method 'items' of 'dict' objects>")
  175.         # XXX member descriptors
  176.         # XXX attribute descriptors
  177.         # XXX slot descriptors
  178.         # static and class methods
  179.         class C:
  180.             def foo(cls): pass
  181.         x = staticmethod(C.foo)
  182.         self.failUnless(repr(x).startswith('<staticmethod object at 0x'))
  183.         x = classmethod(C.foo)
  184.         self.failUnless(repr(x).startswith('<classmethod object at 0x'))
  185.  
  186. def touch(path, text=''):
  187.     fp = open(path, 'w')
  188.     fp.write(text)
  189.     fp.close()
  190.  
  191. def zap(actions, dirname, names):
  192.     for name in names:
  193.         actions.append(os.path.join(dirname, name))
  194.  
  195. class LongReprTest(unittest.TestCase):
  196.     def setUp(self):
  197.         longname = 'areallylongpackageandmodulenametotestreprtruncation'
  198.         self.pkgname = os.path.join(longname)
  199.         self.subpkgname = os.path.join(longname, longname)
  200.         # Make the package and subpackage
  201.         os.mkdir(self.pkgname)
  202.         touch(os.path.join(self.pkgname, '__init__'+os.extsep+'py'))
  203.         os.mkdir(self.subpkgname)
  204.         touch(os.path.join(self.subpkgname, '__init__'+os.extsep+'py'))
  205.         # Remember where we are
  206.         self.here = os.getcwd()
  207.         sys.path.insert(0, self.here)
  208.  
  209.     def tearDown(self):
  210.         actions = []
  211.         os.path.walk(self.pkgname, zap, actions)
  212.         actions.append(self.pkgname)
  213.         actions.sort()
  214.         actions.reverse()
  215.         for p in actions:
  216.             if os.path.isdir(p):
  217.                 os.rmdir(p)
  218.             else:
  219.                 os.remove(p)
  220.         del sys.path[0]
  221.  
  222.     def test_module(self):
  223.         eq = self.assertEquals
  224.         touch(os.path.join(self.subpkgname, self.pkgname + os.extsep + 'py'))
  225.         from areallylongpackageandmodulenametotestreprtruncation.areallylongpackageandmodulenametotestreprtruncation import areallylongpackageandmodulenametotestreprtruncation
  226.         eq(repr(areallylongpackageandmodulenametotestreprtruncation),
  227.            "<module '%s' from '%s'>" % (areallylongpackageandmodulenametotestreprtruncation.__name__, areallylongpackageandmodulenametotestreprtruncation.__file__))
  228.         eq(repr(sys), "<module 'sys' (built-in)>")
  229.  
  230.     def test_type(self):
  231.         eq = self.assertEquals
  232.         touch(os.path.join(self.subpkgname, 'foo'+os.extsep+'py'), '''\
  233. class foo(object):
  234.     pass
  235. ''')
  236.         from areallylongpackageandmodulenametotestreprtruncation.areallylongpackageandmodulenametotestreprtruncation import foo
  237.         eq(repr(foo.foo),
  238.                "<class '%s.foo'>" % foo.__name__)
  239.  
  240.     def test_object(self):
  241.         # XXX Test the repr of a type with a really long tp_name but with no
  242.         # tp_repr.  WIBNI we had ::Inline? :)
  243.         pass
  244.  
  245.     def test_class(self):
  246.         touch(os.path.join(self.subpkgname, 'bar'+os.extsep+'py'), '''\
  247. class bar:
  248.     pass
  249. ''')
  250.         from areallylongpackageandmodulenametotestreprtruncation.areallylongpackageandmodulenametotestreprtruncation import bar
  251.         # Module name may be prefixed with "test.", depending on how run.
  252.         self.failUnless(repr(bar.bar).startswith(
  253.             "<class %s.bar at 0x" % bar.__name__))
  254.  
  255.     def test_instance(self):
  256.         touch(os.path.join(self.subpkgname, 'baz'+os.extsep+'py'), '''\
  257. class baz:
  258.     pass
  259. ''')
  260.         from areallylongpackageandmodulenametotestreprtruncation.areallylongpackageandmodulenametotestreprtruncation import baz
  261.         ibaz = baz.baz()
  262.         self.failUnless(repr(ibaz).startswith(
  263.             "<%s.baz instance at 0x" % baz.__name__))
  264.  
  265.     def test_method(self):
  266.         eq = self.assertEquals
  267.         touch(os.path.join(self.subpkgname, 'qux'+os.extsep+'py'), '''\
  268. class aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa:
  269.     def amethod(self): pass
  270. ''')
  271.         from areallylongpackageandmodulenametotestreprtruncation.areallylongpackageandmodulenametotestreprtruncation import qux
  272.         # Unbound methods first
  273.         eq(repr(qux.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.amethod),
  274.         '<unbound method aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.amethod>')
  275.         # Bound method next
  276.         iqux = qux.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa()
  277.         self.failUnless(repr(iqux.amethod).startswith(
  278.             '<bound method aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.amethod of <%s.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa instance at 0x' \
  279.             % (qux.__name__,) ))
  280.  
  281.     def test_builtin_function(self):
  282.         # XXX test built-in functions and methods with really long names
  283.         pass
  284.  
  285. class ClassWithRepr:
  286.     def __init__(self, s):
  287.         self.s = s
  288.     def __repr__(self):
  289.         return "ClassWithLongRepr(%r)" % self.s
  290.  
  291.  
  292. class ClassWithFailingRepr:
  293.     def __repr__(self):
  294.         raise Exception("This should be caught by Repr.repr_instance")
  295.  
  296.  
  297. def test_main():
  298.     run_unittest(ReprTests)
  299.     if os.name != 'mac':
  300.         run_unittest(LongReprTest)
  301.  
  302.  
  303. if __name__ == "__main__":
  304.     test_main()
  305.